/*
 * Copyright:  2018 smarabbit studio.
 *
 *  Licensed under the Confluent Community License; you may not use this file
 *  except in compliance with the License.  You may obtain a copy of the License at
 *
 *  http://www.confluent.io/confluent-community-license
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 *  WARRANTIES OF ANY KIND, either express or implied.  See the License for the
 *  specific language governing permissions and limitations under the License.
 *
 *  @作   者： 黄开晖 (117227773@qq.com)
 *  @日   期:  2021年8月16日
 */
package com.massyframework.beanskin.sample.plugin;

import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.Instrumentation;

import com.massyframework.beanskin.Environment;
import com.massyframework.beanskin.framework.Framework;
import com.massyframework.beanskin.plugin.Plugin;
import com.massyframework.beanskin.plugin.PluginActivator;

/**
 * @author huangkh
 *
 */
public class SamplePluginActivator implements PluginActivator {
	
	private Framework framework;
	private FragmentMonitor monitor;
	
	
	/**
	 * 
	 */
	public SamplePluginActivator() {

	}

	@Override
	public synchronized void init(Environment<Plugin> env) throws Exception {
		if (this.framework == null) {
			this.framework = env.getFramework();
			
			env.addAwaitStrategy( builder -> {
				builder.addDependencyService(Instrumentation.class)
					.action( repo -> {
						Instrumentation inst = repo.getService(Instrumentation.class);
						ClassFileTransformer transformer = new ByteCodeClassTransformer();
						inst.addTransformer(transformer);
					});
			});
			this.monitor = new FragmentMonitor();
			this.framework.addListener(this.monitor);
		}
	}

	@Override
	public synchronized void destroy() {
		if (this.framework != null) {
			this.framework.removeListener(this.monitor);
			this.monitor = null;
			this.framework = null;
		}
	}

}
